#!/bin/bash
#SBATCH -J frcnn
#SBATCH -p gpu
#SBATCH -N 1
#SBATCH --output=log.out
#SBATCH --error=log.err
#SBATCH --cpus-per-task=8
#SBATCH --gres=gpu:2
. ~/.bashrc
module load cuda/9.2
#1568804161
module load anaconda3/2019.07
#1568804205
conda activate mxnet

python train_faster_rcnn_cus.py --gpus 0,1 --network resnet50_v1b --use-fpn -j 2
